package com.aiolos.springboot.stepbystep.filter;

import com.alibaba.druid.support.http.StatViewServlet;
import com.alibaba.druid.support.http.WebStatFilter;
import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.boot.web.servlet.ServletRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * 使用代码注册Servlet和Filter的方式，开启Druid监控的配置类
 *
 */
@Configuration
public class DruidConfiguration {

    @Bean
    public ServletRegistrationBean druidStatViewServlet(){
        //由于 SpringBoot 默认是以 jar 包的方式启动嵌入式的 Servlet 容器来启动 SpringBoot 的 web 应用，
        //没有 web.xml 文件,所以要使用 Servlet 功能，就必须要借用 Spring Boot 提供的 ServletRegistrationBean 接口
        //使用ServletRegistrationBean 接口实现Servlet的注册
        ServletRegistrationBean servletRegistrationBean =
                new ServletRegistrationBean(new StatViewServlet(),"/druid/*");

        //添加初始化参数
        //设置访问数据库的IP白名单
        servletRegistrationBean.addInitParameter("allow","127.0.0.1");

        //设置访问数据库的IP黑名单 (存在共同时，deny优先于allow)
        // 如果满足deny的话提示:Sorry, you are not permitted to view this page.
        servletRegistrationBean.addInitParameter("deny","192.168.1.73");

        //登录查看信息的账号和密码
        servletRegistrationBean.addInitParameter("loginUsername","admin");
        servletRegistrationBean.addInitParameter("loginPassword","123456");

        //是否能够重置数据
        servletRegistrationBean.addInitParameter("resetEnable","false");

        return servletRegistrationBean;
    }

    @Bean
    public FilterRegistrationBean druidStatFilter(){
        //使用FilterRegistrationBean接口实现Filter的注册
        FilterRegistrationBean filterRegistrationBean
                = new FilterRegistrationBean(new WebStatFilter());

        //添加过滤规则
        filterRegistrationBean.addUrlPatterns("/*");

        //添加需要忽略的格式信息
        filterRegistrationBean.addInitParameter("exclusions",".js,*.gif,*jpg,*.png,*.css,*.ico,/druid/*");

        return filterRegistrationBean;
    }
}
